import pandas as pd
import os

def calculate_hours():
    try:
        # 直接在excel目录下读取文件
        files = [f for f in os.listdir('excel') if f.endswith('.xlsx')]
        if not files:
            print("excel目录下没有Excel文件22")
            return
            
        print(f"找到Excel文件: {files[0]}")
        
        # 读取Excel文件（使用正确的路径）
        file_path = os.path.join('excel', files[0])
        df = pd.read_excel(file_path)
        
        # 检查工时（天）列是否存在
        if '工时（天）' not in df.columns:
            print("未找到'工时（天）'列")
            print(f"可用的列名: {list(df.columns)}")
            return
        
        # 过滤掉NaN值
        valid_hours = df[df['工时（天）'].notna()]
        
        # 打印每一行的工时数据（仅显示有效数据）
        print("\n每行工时详情:")
        for index, row in valid_hours.iterrows():
            print(f"第{index+1}行: {row['工时（天）']} 天")
            
        # 计算工时总和（使用有效数据）
        total_hours = valid_hours['工时（天）'].sum()
        print(f"\n工时统计结果:")
        print(f"总工时（天）: {total_hours:.2f}")
        print(f"有效数据行数: {len(valid_hours)}")
        
    except Exception as e:
        print(f"错误: {str(e)}")

# 直接调用函数
calculate_hours() 